自强化集体动力学:无监督RLVR中跨任务耦合如何驱动策略崩塌
🧭 RLVR自强化训练崩塌机制
Columbia University
揭示无监督RLVR多数投票自奖励的集体不稳定性,解释训练任务数增加导致模型答案趋同崩塌
📌 工作参考价值(3 点)
- 在电商/推荐场景用自一致性投票或 LLM 自评做无监督 RLVR 时,跨任务共享参数会导致策略崩塌到统一输出;应按 task 或 category 拆分 head/梯度,或显式移除共享方向,避免模型对所有 query 给出相同改写/推荐。
- 任务数越多越容易崩塌。多品类、多目标 RLVR 训练时监控自评 reward 与真实 accuracy 的背离,设置验证集早停;任务规模大时可加大 per-task sharpening 的隔离(如 LoRA per task)降低共享耦合 λ。
- 初始模型置信度低时崩塌风险高。业务冷启动做 RLVR 前先让 LLM 在监督数据上收敛到较自信策略,或用 KL/熵正则项(marginal-entropy bonus)稳定 uniform collective mode,防止退化。
大规模流语言模型:连续流实现并行生成与自推测加速
🧭 连续流语言模型 · 块并行解码加速
UC Berkeley · Amazon · KAIST · Carnegie Mellon University
Lafa 8B 连续流语言模型用 <30B token 训练达到 AR 质量,并以自推测解码实现 2.54 倍吞吐
📌 工作参考价值(4 点)
- 连续流语言模型可直接适配预训练 AR 模型,只用不到 30B token 训练即可达到匹配或更好的指令跟随/推理/编码质量,对业务中已有 LLM 的低成本改造有参考价值;在搜索 query 生成、广告文案生成等高吞吐场景,可将 AR 模型转为流式并行解码。
- 自推测解码方案用同一模型生成流式草稿并自回归验证,无需额外 draft 模型,相较 multi-token prediction 和离散扩散方法吞吐提升显著(2.54 倍),可直接用于线上推理降本。
- blockwise 流建模 + prefix caching + 块并行解码的组合可以作为工程参考:前缀因果注意力保证与 AR 的兼容性,块内双向异步流提升并行度,适合生成式推荐的 prompt 长、重复前缀多的场景。
- 流训练带来的 multi-token prediction 效应能保留甚至提升预训练 AR 能力,说明可以在业务微调阶段引入流训练做能力增强,而不只是推理加速。
Waypoint-OPD:基于门控在线策略蒸馏的 RL 策略偏移迁移
🧭 RL 策略偏移蒸馏 · 门控迁移
Alibaba Cloud Computing
用中间检查点方向一致性门控教师 RL 前后策略偏移,提升数学与代码推理蒸馏效果
📌 工作参考价值(4 点)
- 在 RL 蒸馏中保留中间 checkpoint 成本低:只需额外一次 frozen-teacher 打分,即可按 early/late 偏好方向一致性过滤噪声,适合线上小模型蒸馏。
- 门控思想可迁移到排序/Agent 策略蒸馏:teacher 策略在训练中迭代不稳定时,不要直接用最终参数差,而是用中间检查点检测方向冲突,抑制相互抵消的 policy shift。
- 对用 RL 微调 LLM 做 query 生成、生成式推荐或 Agent 决策的业务,可以借鉴 on-policy 蒸馏方式:让学生自生成轨迹,teacher 打分并计算 early/late log prob 差,只把方向一致的策略偏移作为监督,避免把探索期噪声传给学生。
- 论文在数学与代码推理验证,迁移到电商场景需适配奖励和动作空间,但“方向一致性 gating”是通用 trick。
预测策略内自蒸馏中特权信息的有效性
🧭 多轮代理场景下的自蒸馏训练
University of Michigan – Ann Arbor · Mila · Scale AI
系统评估多轮代理场景下16种特权信息配置,提出KL散度指标预测其训练收益
📌 工作参考价值(4 点)
- PI 的内容与生成源比固定设计更重要:优先选择信息密度高、由更强模型生成的轨迹片段作为 PI,尤其在多轮 Agent 对话中可显著提升自蒸馏收益。
- 训练前用 KL 指标预筛 PI 候选:计算 sampled-token reverse KL (D) 与 error-step KL contrast (Δd^{-}_{err}),比直接看 PI utility 更能稳定预测训练后性能,可避免信息泄漏导致的误判。
- 当候选 PI 差异较大时,利用 Δd^{-}_{err} 为每个任务动态选择 PI,比随机选择平均提升更高;适合业务中多任务、多轨迹来源的场景。
- 自蒸馏目标可与 GRPO 等强化学习目标结合,在有效 PI 下获得比纯 GRPO 更强的性能,适合需要兼顾模仿与探索的电商 Agent 训练。
谱驱动自适应秩优化器,用于可验证奖励的 LLM 强化学习
🧭 LLM RLVR 优化器设计
Gaoling School of Artificial Intelligence, Renmin University of China
发现 RLVR 梯度能量向低秩子空间集中,提出自适应秩优化器 Ares,跨模型和任务提升准确率
📌 工作参考价值(3 点)
- 在业务中用 RLVR 微调推荐/搜索 Agent 时,可借鉴 Ares 的自适应低秩优化:根据梯度能量占比动态选择更新秩,避免固定 LoRA rank 的欠拟合/过拟合,同时降低显存与通信开销。
- 监控梯度谱能量分布与有效秩下降趋势,可作为 RLVR 训练健康度的诊断指标;若梯度能量没有像预期那样向低秩集中,可能说明奖励信号或任务设计有问题,便于早停或调整。
- 实现上,Ares 的能量阈值法(估计保留一定能量比例的最小秩)可以复用到现有多卡训练框架中,对优化器状态做低秩投影,提升训练吞吐,适合大规模电商搜索推理模型更新。
超越领域路由监督:多教师在线蒸馏的跨教师守护
🧭 多教师在线蒸馏梯度冲突缓解
Nanjing University · JD.com
提出 ROADMOPD,引入非路由教师作为跨域批评者,通过 token 级信用分配与梯度投影缓解多教师在线蒸馏的梯度冲突
📌 工作参考价值(4 点)
- 在多领域/多任务 LLM 在线蒸馏或 RLHF 中,不要只靠数据路由区分教师;可在同一批轨迹上额外计算非路由教师的梯度/损失作为冲突检测器,用教师一致性加权 token 级损失,低成本缓解共享参数下的梯度冲突。
- 梯度投影是可复用的训练工程技巧:当辅助梯度与主梯度方向冲突时,对主更新做投影去除冲突分量;但落地时应自适应控制投影强度,避免过度约束损害目标域信号,可参考其投影强度的门控/衰减机制。
- 若业务中需同时蒸馏多个领域专家(如电商搜索、推荐文案、客服),可借鉴“跨教师守护”思路:把非目标域教师作为 critic,不直接更新学生,只参与冲突感知的梯度修正,降低劣势领域退化风险。
- 对于 token 级信用分配:利用教师间一致性而非单教师概率来调整 token loss 权重,能减少噪声标签/风格差异带来的干扰,适合生成式推荐中文本质量与领域风格冲突的场景。
基于预条件空间的舍入:重新设计4-bit AdamW优化器状态量化
🧭 4-bit AdamW 优化器状态量化
University of California, Berkeley · Nubank
提出在预条件空间做随机舍入的4-bit AdamW状态量化方法,显著缩小与32-bit训练的验证损失差距
📌 工作参考价值(4 点)
- 若已经在用 TorchAO 4-bit AdamW 压缩大模型优化器状态,可直接替换第二动量的量化方式:不再做 state-space rounding,而是在 preconditioner space 计算随机舍入概率,改动集中在量化器实现层,训练流程无需重构。
- 第二动量码书可两条路线选择:ZIP-SR 保留零点适合大多数场景;ZE-EDEN 排除零点并 rescale 量化块,可缓解正量化底带来的预条件失真,对推荐/搜索模型中大量稀疏或接近零的梯度更有针对性。
- 第一动量用 4-bit NF4 已经够稳,但输出层(如推荐系统中的 item embedding 或 LLM 的 LM head)对量化更敏感,最后 10% 训练步对输出层第一动量做针对性随机舍入,能进一步降低验证损失,适合在微调生成式推荐模型时采用。
- 对需要大规模预训练或全参数 SFT 的电商/Agent 场景,该方法可降低优化器显存占用,同时把 4-bit 训练与 32-bit 训练之间的验证损失差距最多缩小 70%,适合作为训练成本优化的默认配置。
一个模型循环所有层:语言模型的弹性潜在循环
🧭 LLM 弹性深度与循环推理降本 ★ 深度解读
NVIDIA
通过 adaptive-delta 更新与 rank override 混合深度训练,循环 LLM 在 1 loop 不损精度并随 loop 增长持续提分
📌 工作参考价值(5 点)
- 单模型多档线上服务:ELR 提供一个 checkpoint 通过 loop count 旋钮控制精度/延迟,可替代为不同 latency tier 维护多个模型;在电商搜索、广告文案生成、Agent 长 rollout 等场景可直接按流量层级切换 1-4 loop,无需重训或增加参数。
- 低成本接入循环 LLM:延迟混合深度训练只在最后 25% token 开启,训练 FLOPs 仅约 +16%;已有非循环模型可通过 continuation 方式获得弹性深度,比从头训练循环模型更划算。
- MoE 推理优化技巧:cross-loop route replay 复用首 loop 专家路由,训练 update time 降低 11.7%,推理 forward 降低 1.2% 且精度几乎不变;对于电商大规模 MoE 模型,这是降低反复 router 计算的直接可抄方案。
- KV cache 与生成成本:last-loop KV caching 让 attention KV 显存保持 1-loop 水平,同时保留大部分深度收益;高 loop 还会缩短生成 tokens,适合长上下文 Agent/多轮对话降低显存与耗时。
- 避免 one-loop tax 的关键:必须用 adaptive-delta 更新和 within-step rank override 混合 1-loop 与多 loop 梯度;纯 synchronized sampling 或加权采样无法保住 1-loop 精度,做循环化改造时要避免直接沿用 Huginn 式训练。
无搜索锐化:序列级幂分布的在线策略蒸馏
🧭 LLM 训练 · 无奖励推理锐化 ★ 深度解读
University of Southern California · Intel AI
OPPD 通过 on-policy 蒸馏把推理时序列级幂分布采样的成本转移到训练阶段,单个生成即可达到或超过 64 候选采样
📌 工作参考价值(4 点)
- 在电商搜索 query 推荐或文案生成中,可以训练模型学习教师模型对完整序列的幂分布,即对完整候选文本(如 query)的概率进行 α 次幂加权,使推理时只需单次生成即可达到近似多个候选采样再加权的效果,大幅降低线上推理延迟。
- 借鉴 on-policy weighted likelihood 训练方式:让待训练模型自己生成候选,用冻结教师(如 LLM)给出概率并计算权重,重点加权学生自身的高概率答案;这比直接蒸馏教师生成样本更贴合实际分布,适用于 Agent 工具调用轨迹或推荐解释文本的训练。
- 工程实现 trick:生成候选时使用教师温度 1/α 的 proposal 降低重要性权重方差,避免权重退化;将长文本分块(例如 64 tokens)让教师一次前向评分,减少调用次数;训练权重与 SMC 重要性权重分别累加,并在重采样后重新归一化。这些都能复用在大规模在线生成任务的稳定训练中。
- 在没有人工偏好或点击信号可用时,可以使用模型自身的序列概率(经由 teacher 锐化)作为软奖励进行训练,提升推理和生成质量;对电商问答、商品推荐理由、营销文案生成等场景可低成本迁移。
散度选择控制蒸馏中的熵变化
🧭 LLM 蒸馏目标的熵效应
Stanford University
证明 forward KL 抬高学生熵、reverse KL 压低熵,散度成为隐式熵正则器
📌 工作参考价值(4 点)
- 蒸馏生成式推荐或 query 生成模型时,按输出分布需求选散度:forward KL 会抬高学生熵,适合增加候选多样性/探索性;reverse KL 压低熵,适合需要集中、高置信输出的广告文案、push 文案等场景,但需注意 student-teacher 差距过大时 reverse KL 会欠拟合。
- on-policy 蒸馏中低熵主要来自 token-level reverse KL,而非 on-policy 采样本身;如果 RL/on-policy 训练后模型行为过于 deterministic,可以直接调整 token 级散度损失,而不必改动采样数据策略。
- 使用 interpolated KL 时,训练早期熵变化平滑、收敛时突变;可在训练中监控熵曲线,在收敛前设置早停或动态温度,避免熵崩溃损伤生成多样性。
- self-distillation 中 teacher 拥有特权特征会降低 student 熵,此时应调节散度超参或 softmax 温度做补偿;在电商 teacher-student 迁移中,student 无法使用用户实时行为等特权信息时,可提高 forward KL 权重或降低温度以保持输出多样性。
CorrGRPO:多奖励学习的相关性归一化 GRPO
🧭 多奖励 RL · GRPO 归一化 ★ 深度解读
Hong Kong University of Science and Technology
将 GRPO 多奖励优势归一化从协方差改为 Pearson 相关,消除大方差奖励主导,提升代码/工具/Agent 安全多目标训练效果
📌 工作参考价值(4 点)
- 工程实现简单:只需在 group advantage 分母中用相关矩阵元素和替换协方差和,分子保持 centered total reward;论文 Listing 1 可直接作为现有多奖励 GRPO 训练代码的改造模板。
- 当业务奖励量纲/方差差异大(例如 CTR、转化、时长、多样性、安全分)时,CorrGRPO 可避免大方差目标主导优势缩放,同时保留预设目标权重;适合作为多目标 RL 的默认 advantage normalizer。
- 高度相关的奖励(点击与转化、多个可用性指标)会被相关归一化自动去冗余,避免重复奖励放大更新;在 reward hacking 或目标一致性强时尤其有用,可辅助探索。
- 训练动态上 CorrGRPO 保留更高策略熵,减缓 entropy collapse,有利于持续探索和发现高奖励策略;可与 KL 系数、DAPO/CISPO 等现有 RL 方法结合。
分数中心化稳定离线策略强化学习
🧭 LLM 强化学习训练稳定性校正
Together AI
提出加性 score centering 校正项,抵消训练与推理引擎间的漂移,稳定 LLM 的 RL 训练
📌 工作参考价值(4 点)
- 在 RL 微调推荐/对话策略时,若部署推理引擎与训练引擎存在量化、KV cache 压缩或异步参数不一致,可在计算优势或回报时减去漂移项(如训练与推理引擎在同样本上的平均输出差异),低成本稳定训练,避免策略崩溃。
- Score centering 是加性修正,可与 importance sampling 等乘法修正叠加;在离线策略数据来自旧策略或不同推理引擎时,组合使用可能比单用 IS 更稳健,尤其 mismatch 严重时。
- 对于生成式推荐(用 LLM 直接生成 item ID/文案),RL 训练中 reward 常由规则或模型给出,容易受采样分布偏移影响;考虑在 reward 或 advantage 上做 centering 来抵消环境漂移,可提升训练稳定性。
- 工程上:漂移估计可通过在固定验证集上比较训练和推理引擎的 logprob/奖励差异得到,开销很小;可作为 RL 训练 pipeline 的常规监控与校正组件。
学习自己的思考:抽象 Token 课程
🧭 课程学习训练隐式思维表示
UC Berkeley · UC San Diego · EPFL · Google DeepMind
提出抽象 Token 课程(ATC),无需显式思维链监督,通过课程学习在连续表示空间形成内部抽象思考
📌 工作参考价值(4 点)
- 在训练推荐/Agent 的规划与推理能力时,可采用课程学习思路:按任务难度(如多跳路径长度、图规模、算术复杂度)从易到难组织样本,让模型在连续隐藏状态中隐式形成推理步骤,省去逐条标注 CoT 的成本。
- 若业务需要多步推理但显式 CoT 延迟高或标注贵,可尝试在 Transformer 中间层引入可训练的抽象 token,配合渐进课程,将中间推理压缩到连续表示空间,降低推理时的 token 开销。
- 理论分析提示注意力会自然聚焦于上下文中提供“最容易路径”的 token;构造课程样本时可在早期保留关键捷径 token,帮助模型更快收敛,后期再逐步移除捷径以强化内部推理。
- 当前验证集中在图可达性与算术任务,属理论性实验;迁移到大规模电商/推荐场景前,需要先在真实多跳推理任务上做小规模验证。
当 EOS Token 不一致:理解 On-Policy 蒸馏中的长度膨胀
🧭 On-policy 蒸馏终止 token 失配 ★ 深度解读
University of North Carolina at Chapel Hill · Brigham Young University · Microsoft
发现 base 学生与 post-trained 教师的终止 token 失配是 OPD 长度膨胀主因,语义 EOS 聚合对齐终止概率可有效缓解
📌 工作参考价值(4 点)
- 在电商/推荐/Agent 场景用小模型蒸馏大模型生成推荐理由、query 改写或对话回复时,务必检查学生和教师模型的 EOS token 配置及实际概率分布;即使声明停止集相同,也可能分别偏好 <eos> 与 <end_of_turn>,导致生成不停止、重复输出。
- 修复不能只改 decoding 的 stopping set,要在训练目标层把语义等价的终止 token 概率求和,作为同一个 stop 动作进行监督(semantic EOS aggregation);否则学生已有的停止动作会被负梯度持续压制,且教师偏好的 token 因采样概率极低几乎得不到直接监督。
- OPD 训练中应监控 response length、clipping ratio、termination probability 三条曲线;若正确答案早已出现但长度继续增长、终止概率骤降,优先怀疑 EOS 失配,而不是先调学习率或换 KL 目标。
- 多轮 Agent 或工具调用场景不要盲目聚合所有特殊 token:需定义 context-dependent termination classes,区分 end-of-turn、tool call、document boundary、message completion,否则会破坏交互控制流。
D^3-MOPD:自适应动态领域调度的高效多教师蒸馏
🧭 多教师蒸馏 · 在线动态调度
AllSpark Team
提出零开销动态调度器,复用每领域reverse-KL信号在线调整多教师蒸馏的数据混合比例,大幅提升收敛效率与性能
📌 工作参考价值(4 点)
- 借鉴 per-domain loss/KL 轨迹作为在线调度信号:在多任务/多域推荐模型(如首页推荐、搜索、广告域联合训练)中,定期追踪各任务的收敛速度与剩余提升空间,动态调整采样比例,避免固定混合导致快任务过拟合、慢任务欠训练。
- 使用 off-process 异步 watcher 实现零开销调度:不侵入训练主循环,适合在现有训练框架中轻量集成,无需改动核心代码即可获得收益。
- 调度策略可迁移到多场景 LLM 微调:当不同领域数据收敛速度差异大时,优先分配资源给“难但可学”的领域,减少已饱和领域的采样,能显著降低达到目标性能所需的训练步数。
- 结论“领域越多、收敛模式差异越大,动态调度收益越大”提示:在电商多域多目标场景下,随着接入业务域增多,这类自适应调度可能带来超线性训练效率提升。
ALiBi 位置编码的浮点下溢失明问题与缓解策略
🧭 ALiBi 位置编码数值缺陷分析与缓解
Leipzig University (InfAI) · ScaDS.AI Dresden/Leipzig · Seltz · University of Kassel · hessian.AI
发现 ALiBi 的线性偏置在长距离下因浮点下溢导致注意力权重为零,提出 log 缩放距离等缓解方法
📌 工作参考价值(4 点)
- 若在电商搜索推荐等场景使用基于 ALiBi 的 decoder-only LLM(如生成 query、长用户序列建模),需注意长距离 token 的注意力权重可能因浮点下溢变为零,导致头部“失明”,影响长文本检索能力。可通过监控 attention underflow 比例来诊断。
- 训练时可改用 log-scaled distances 作为位置偏置,该方法在 passkey retrieval 任务中表现最稳定,能显著缓解下溢;也可结合 clamping 或混合偏置。
- 评估长上下文模型时,不能仅看常规困惑度或标准基准,需加入 needle-in-a-haystack 类 token 检索任务,避免 ALiBi 的隐性缺陷影响线上长序列性能。
- 若必须沿用默认 ALiBi 斜率,应意识到其作为强基线在 needle-in-a-haystack 任务上仍有竞争力,但需权衡训练稳定性与检索能力损失。
通过平滑 MMD 对齐增强大语言模型的数值预测
🧭 LLM 数值预测 · 分布对齐训练损失
College of Computer Science, Sichuan University, Chengdu, China · Dongfang Electric (Chengdu) Academy of Science and Technology Co., Ltd., Chengdu, China · Institute of Medical Information & Library, CAMS & PUMC, Beijing, China
提出 SMMD 损失,用数值距离核与图平滑将对齐预测分布,一致提升 LLM 数值精度
📌 工作参考价值(4 点)
- 数值敏感推荐场景损失替代:在需精确数值的任务(如价格预估、销量预测、广告出价、用户评分)中,可将标准交叉熵替换为 SMMD,将输出视为连续分布对齐而非离散 token 分类,减少“数字错位”错误。
- 数值子词表与距离核设计:显式划分数值 token 子词表,构造基于值距离的 RBF 核,将 token 概率分布映射到数值空间进行 MMD 匹配。电商搜索排序中的分数校准、点击率预估等场景可直接复用此核构建策略。
- 局部平滑正则引入:利用核图在相邻数值间施加输出残差平滑约束,抑制跳变预测。推荐系统中用户偏好分数、物品特征值等的生成,可借鉴该平滑项提升预测一致性与鲁棒性。
- 即插即用训练组件:SMMD 仅修改损失项,不改变模型结构,可轻量集成到现有 LLM fine-tune 流程中,尤其适合需要数值输出的 Agent 决策(如动态定价、预算分配)。
分层Muon:高效训练的分块Newton-Schulz更新
🧭 高效训练优化器 · 分块Newton-Schulz
通过将动量梯度矩阵分块独立应用Newton-Schulz,大幅降低Muon优化器计算量并实现高效GPU并行
📌 工作参考价值(4 点)
- 在训练含有大型密集矩阵参数的推荐或广告模型时,可将 Muon 优化器的 Newton-Schulz 步改为分块方案,直接降低从 O(r²s) 到 O(HWT) 的单步计算量,同时保持接近全矩阵更新的训练效果。
- 利用独立小方块计算的特性,可以设计 tile-size 相关的 GPU kernel,配合跨层批处理(cross-layer batching),将不同层的同尺寸块合并批量矩阵乘法,显著提升 GPU 吞吐。
- 采用内存受限分块(memory-bounded chunking)和运行时动态调整块尺寸的策略,能灵活适配不同硬件显存限制,避免 OOM 且不牺牲收敛。
- 该方法将矩阵谱交互限定在块内,可作为工程上折衷效果与效率的通用抽象,类似思路可迁移至其他需要 per‑parameter 矩阵变换的优化场景(如 Shampoo 类优化器的分块近似)。
通过能力选择性子空间投影的自策略蒸馏
🧭 LLM 自蒸馏 · 能力选择性子空间
University of Cambridge · HKUST · University of Chicago
无需外部信号的自我蒸馏方法,利用低秩能力子空间投影筛选自生成数据,显著提升LLM特定能力
📌 工作参考价值(4 点)
- 在无外部信号(如标注数据、奖励模型)的场景下,可通过能力子空间投影从模型自身梯度中提取任务相关特征,实现自我蒸馏,适用于电商对话、推荐理由生成等任务的低成本优化。
- 利用 KV 缓存投影到低秩能力子空间,相当于在自生成过程中隐式施加“能力滤波器”,可稳定提升特定能力(如商品推荐的合理性、数学推理),避免无关特征干扰。
- 方法具备良好的域外泛化性(15% 提升),可跨任务迁移,适合在电商多任务场景下统一提升模型的多方面能力。
- 工程实现上,只需在自生成前向过程中插入一次投影操作,与标准下一 token 预测损失结合,无需修改训练目标,易于集成到现有 LLM 微调流程中。
PlanningBench:面向LLM的可扩展、可验证规划数据生成框架
🧭 生成式规划数据 · 可控难度与自动验证 ★ 深度解读
Renmin University of China · Tencent Hunyuan Team · Beijing Academy of Artificial Intelligence · The Chinese University of Hong Kong
从真实场景抽象出任务-约束分类,通过闭环合成生成可验证规划数据,兼顾评估与强化学习训练
📌 工作参考价值(4 点)
- 可控数据生成范式:约束驱动合成 pipeline 从真实场景抽象任务-约束分类,可迁移至电商复杂任务(如物流调度、大促排期、库存分配)的规划数据生成,突破固定基准的限制。
- 可验证奖励设计:每个实例绑定自动验证清单,同时检查约束满足与目标优化,可作为强化学习 reward,保证 Agent 在长序列决策中保持全局一致性。
- 奖励确定性至关重要:偏好确定最优解,避免多个可行解导致奖励信号分散;在训练多目标协调的 Agent 时,应优先构造明确的最优指标以稳定训练。
- 难度自适应分级:按约束层级(基础/中等/困难)和状态化约束分档,通过动态调整采样概率实现难度递增,适合生成难度可控的训练集。
运行时编排的二阶优化实现可扩展大模型训练
🧭 二阶训练运行时优化
University of Oxford
Asteria通过异步状态分布和有界陈旧同步,将二阶优化器高效应用于大模型训练
📌 工作参考价值(4 点)
- 优化器状态卸载到主机:将二阶优化器的大矩阵状态移至CPU内存或NVMe,GPU仅保留模型参数与梯度,大幅降低显存占用,使推荐/Agent模型中亦可尝试SOAP等二阶优化器以提升样本效率。
- 异步逆平方根计算:利用训练hook预准备影子状态,在主机端后台计算昂贵的矩阵逆平方根,GPU不用等待,适合大embedding矩阵等场景下将耗时的矩阵运算剥离出关键路径。
- 有界陈旧同步协议:分布式训练中设定同步频率上限,结合拓扑感知协调,在多个worker间减少通信开销的同时保障优化有效性,可直接在电商推荐模型的分布式训练中推广。
- 存储分层与动态调度:根据运行时内存压力,动态在GPU显存、CPU内存、NVMe间迁移优化器状态,应对内存受限环境,为部署大规模agent或生成式推荐模型的训练提供弹性资源管理思路。